0%

在有序数组中插入数字

题目要求:

给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。

示例:

输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3

输出: [1,2,2,3,5,6]

将nums1作为输出的数组,则首先将nums1备份,然后进行双指针循环,在两个数组中找最小数字插入到nums1中。因为两个数组中的数字是排序好的,所以最后由一个数组有空余,就将它加入到输出数组的末尾。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution:
def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None:
nums1_copy=nums1[:m]
nums1[:]=[]
p=0
q=0
i=0
while p<m and q<n:
if nums1_copy[p]<nums2[q]:
nums1.append(nums1_copy[p])
p=p+1
else:
nums1.append(nums2[q])
q=q+1
i=i+1
if p<m:
nums1[p+q:]=nums1_copy[p:]
elif q<n:
nums1[p+q:]=nums2[q:]
print(nums1)